# mypy completion                                          -*- shell-script -*-

_mypy()
{
    local cur prev words cword split
    _init_completion -s || return

    [[ $cword -gt 2 && ${words[cword - 2]} == --shadow-file ]] &&
        prev=--shadow-file # hack; takes two args

    case $prev in
        --help | --version | --python-version | --platform | --always-true | \
            --always-false | --@(dis|en)able-error-code | --find-occurrences | \
            --exclude | --package | --command | -!(-*)[hVpc])
            return
            ;;
        --config-file)
            _filedir
            return
            ;;
        --follow-imports)
            COMPREPLY=($(compgen -W 'normal silent skip error' -- "$cur"))
            return
            ;;
        --python-executable)
            COMPREPLY=($(compgen -c -- "${cur:-py}"))
            return
            ;;
        --*-dir | --*-report)
            _filedir -d
            return
            ;;
        --custom-typing-module | --module | -!(-*)m)
            _xfunc python _python_modules
            return
            ;;
        --shadow-file)
            _filedir '@(py|pyi)'
            return
            ;;
        --junit-xml)
            _filedir xml
            return
            ;;
    esac

    $split && return

    if [[ $cur == -* ]]; then
        COMPREPLY=($(compgen -W '$(_parse_help "$1")' -- "$cur"))
        return
    fi

    _filedir '@(py|pyi)'
} &&
    complete -F _mypy mypy

# ex: filetype=sh
